package com.king.netty.groupchat;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.text.DateFormatSymbols;

/**
 * @author unidentifiable
 * @date 2021-03-30
 */
public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {

    /**
     * 定义一个 channel 组，管理所有的 channel
     * GlobalEventExecutor.INSTANCE 是全局的事件执行器，是一个单例对象
     */
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
//    private static final DateFormatSymbols dfs = new DateFormatSymbols("yyyy-MM-dd HH:mm:ss");

    /**
     * handlerAdded 表示建立连接，一旦连接，第一个被执行
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        // 将该客户加入聊天的信息推送给其他在线的客户端
        channelGroup.writeAndFlush(channel.remoteAddress() + " 加入了聊天室" + "\n");
        channelGroup.add(channel);
//        super.handlerAdded(ctx);
    }

    /**
     * handlerRemoved 断开连接，从 channelGroup 中删除
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        channelGroup.writeAndFlush(ctx.channel().remoteAddress() + " 离开了聊天室" + "\n");
        System.out.println("当前人数：" + channelGroup.size());
//        super.handlerRemoved(ctx);
    }

    /**
     * 表示channel 处于活跃状态
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        System.out.println(channel.remoteAddress() + " 上线了~");
//        super.channelActive(ctx);
    }

    /**
     * 表示channel 处于非活跃状态
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        System.out.println(channel.remoteAddress() + " 下线了~");
//        super.channelInactive(ctx);
    }

    /**
     * 消息处理
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        // 获取当前channel
        Channel channel = ctx.channel();
        // 遍历当前组
        channelGroup.forEach(ch -> {
            if(channel != ch){
                ch.writeAndFlush("[" + channel.remoteAddress() + "] 对你说：" + msg + "\n");
            }else{
                ch.writeAndFlush("[自己]：" + msg + "\n");
            }
        });
    }

    /**
     * 异常处理
     * @param ctx
     * @param cause
     * @throws Exception
     */
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        ctx.close();
//        super.exceptionCaught(ctx, cause);
    }
}
